"""E2E check for ModalProvider against real Modal infra (echo_env)."""

  import functools

  import modal
  import requests

  # --- WORKAROUND: Modal's encrypted tunnel doesn't forward WS ping/pong frames,
  # so disable the websockets client keepalive for tunnelled connections. ---
  import openenv.core.env_client as _ec
  _ec.ws_connect = functools.partial(_ec.ws_connect, ping_interval=None)

  from openenv.core.containers.runtime.modal_provider import ModalProvider

  DOCKERFILE = "envs/echo_env/server/Dockerfile"   # path relative to repo root


  def run(use_v2: bool) -> None:
      print(f"\n=== ModalProvider e2e (use_sandbox_v2={use_v2}) ===")

      # ---- provider config ----
      provider = ModalProvider(
          app_name="openenv-e2e",     # Modal app (created if missing)
          use_sandbox_v2=use_v2,      # False -> Sandbox.create, True -> _experimental_create
          timeout=600,                # max sandbox lifetime (s)
          cpu=1.0,                    # optional
          memory=1024,                # optional; ignored when use_sandbox_v2=True
          # cmd="cd /app/env && uvicorn server.app:app --host 0.0.0.0 --port 8000",  # optional override
      )

      image = ModalProvider.image_from_dockerfile(DOCKERFILE)  # context defaults to envs/echo_env/
      print("image:", image)

      with modal.enable_output():                 # stream the build logs
          base_url = provider.start_container(image)
      print("base_url:", base_url)

      try:
          provider.wait_for_ready(base_url, timeout_s=180)
          print("health ready ✅")

          r = requests.get(f"{base_url}/health", timeout=10)
          assert r.status_code == 200, r.status_code
          print(f"GET /health -> {r.status_code} ✅")

          from echo_env.client import EchoEnv
          with EchoEnv(base_url=base_url).sync() as env:   # .sync() — client is async by default
              result = env.reset()
              print("reset ->", type(result).__name__, "✅")
          print(f"MCP round-trip OK (v2={use_v2}) ✅")
      finally:
          provider.stop_container()
          print("sandbox terminated")


  if __name__ == "__main__":
      run(use_v2=False)   # stable Sandbox API
      run(use_v2=True)    # Sandbox v2 (beta)
      print("\nALL E2E CHECKS PASSED ✅")